import { promises as fs } from 'node:fs';
import { join, normalize } from 'node:path';

const rootDir = process.cwd();

/**
 * 递归查找并删除目标目录
 * @param {string} currentDir - 当前遍历的目录路径
 * @param {string[]} targets - 要删除的目标列表
 */
async function cleanTargetsRecursively(currentDir, targets) {
    const items = await fs.readdir(currentDir);

    for (const item of items) {
        try {
            const itemPath = normalize(join(currentDir, item));
            const stat = await fs.lstat(itemPath);

            if (targets.includes(item)) {
                // 匹配到目标目录或文件时直接删除
                await fs.rm(itemPath, { force: true, recursive: true });
                console.log(`Deleted: ${itemPath}`);
            } else if (stat.isDirectory()) {
                // 只对目录进行递归处理
                await cleanTargetsRecursively(itemPath, targets);
            }
        } catch (error) {
            console.error(`Error handling item ${item} in ${currentDir}: ${error.message}`);
        }
    }
}

(async function startCleanup() {
    // 要删除的目录及文件名称
    const targets = ['node_modules', 'dist', '.turbo', 'dist.zip'];
    const deleteLockFile = process.argv.includes('--del-lock');
    const cleanupTargets = [...targets];

    if (deleteLockFile) {
        cleanupTargets.push('pnpm-lock.yaml');
    }

    console.log(`Starting cleanup of targets: ${cleanupTargets.join(', ')} from root: ${rootDir}`);

    try {
        await cleanTargetsRecursively(rootDir, cleanupTargets);
        console.log('Cleanup process completed successfully.');
    } catch (error) {
        console.error(`Unexpected error during cleanup: ${error.message}`);
        process.exit(1);
    }
})();
